#!/bin/bash

cd ..
python run_main_channel.py \
    --task_name probe_weight_prediction \
    --is_training 1 \
    --data CHANNEL \
    --root_path ./dataset/channel_data/ \
    --data_path channel_data.npz \
    --model ChannelLLM \
    --model_id Channel_25_32 \
    --features M \
    --loader channel \
    --seq_len 20 \
    --label_len 10 \
    --pred_len 10 \
    --n_clusters 25 \
    --n_probes 32 \
    --d_model 128 \
    --n_heads 8 \
    --d_ff 256 \
    --dropout 0.1 \
    --llm_layers 6 \
    --batch_size 16 \
    --eval_batch_size 8 \
    --learning_rate 0.001 \
    --train_epochs 30 \
    --patience 7 \
    --des Exp \
    --loss kl \
    --optim adamw \
    --model_comment v1.0 